package simple;

import org.junit.Test;

/**
 * 35. 搜索插入位置
 * 给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。
 * <p>
 * 你可以假设数组中无重复元素。
 * <p>
 * 示例 1:
 * <p>
 * 输入: [1,3,5,6], 5
 * 输出: 2
 * 示例 2:
 * <p>
 * 输入: [1,3,5,6], 2
 * 输出: 1
 * 示例 3:
 * <p>
 * 输入: [1,3,5,6], 7
 * 输出: 4
 * 示例 4:
 * <p>
 * 输入: [1,3,5,6], 0
 * 输出: 0
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/search-insert-position
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 *
 * @author Chen Wei
 * @date 2020-11-18 16:10
 */
public class No35SearchInsert {
    // 二分法
    public int searchInsert(int[] nums, int target) {
        int n = nums.length;
        // ans 返回值
        int left = 0, right = n - 1, ans = n;
        while (left <= right) {
            // 计算中点的索引值
            // int mid = (right + left) >>> 1;
            int mid = ((right - left) >> 1) + left;
            // 判断目标值在数组中点的左侧还是右侧
            if (target <= nums[mid]) {
                ans = mid;
                // 将中点左1的索引赋给右极限
                right = mid - 1;
            } else {
                // 将中点右1的索引赋给左极限
                left = mid + 1;
            }
        }
        return ans;
    }

    @Test
    public void test1() {
        System.out.println(searchInsert(new int[]{1, 3, 5, 6}, 7));
    }

    @Test
    public void test2() {
        System.out.println(searchInsert(new int[]{1, 3}, 2));
    }

    /**
     * 二分法查找元素
     *
     * @param nums
     * @param target
     * @return
     */
    public int binarySearch(int[] nums, int target) {
        // 初始化 下限、上限
        int left = 0, right = nums.length - 1;
        // 若下限 小于等 上限，则循环
        while (left <= right) {
            // 计算 中点索引值，>>> 无符号右移1位，防止溢出 FIXME 防止什么溢出？
            int mid = (right + left) >>> 1;
            if (target == nums[mid]) {
                return mid;
            } else if (target < nums[mid]) {
                right = mid - 1;
            } else {
                left = mid + 1;
            }
        }
        return -1;
    }

    @Test
    public void test3() {
        System.out.println(binarySearch(new int[]{1, 3, 5, 6}, 6));
    }

}
